import os
import re
import time

import requests
from bs4 import BeautifulSoup
from selenium import webdriver


class downloadchromedriver:
    def __init__(self, hd):
        self.hd = hd

    def getHTML(self, url):
        try:
            # 更改requests请求的头信息，用来防止网页的反爬虫，也可来伪装自己访问网页的信息，这里用最简单的chorm（浏览器）版本10代替
            r = requests.get(url, timeout=30, headers=self.hd)
            r.raise_for_status()  # 容错机制，若请求访问失败则返回的不是200，则返回字符串空
            r.encoding = r.apparent_encoding  # 设置编码方式，用解析返回网页源码得出的编码方式代替  UTF-8
            return r.text
        except:
            return ""

    def getHTMLdriver(self, url):
        try:
            # 更改requests请求的头信息，用来防止网页的反爬虫，也可来伪装自己访问网页的信息，这里用最简单的chorm（浏览器）版本10代替
            driver = webdriver.Chrome("D:\python\chromedriver_win32\chromedriver.exe")
            driver.minimize_window()
            driver.get(url)
            time.sleep(3)
            text = driver.page_source
            driver.close()
            return text
        except:
            driver.close()
            return ""

    def save(self, lasturl, dir_name, file_name):
        reponse = requests.get(lasturl)
        dir_name = 'D:/chromedriver/' + dir_name  # 设置文件夹的名字
        # print(dir_name)
        if not os.path.exists(dir_name):  # os模块判断并创建
            os.mkdir(dir_name)
        if not os.path.exists(dir_name + "/" + file_name):
            with open(dir_name + "/" + file_name, 'wb') as f:
                print("downloading:"+dir_name + "/" + file_name)
                f.write(reponse.content)
                print("download success")

    def download(self, url):
        try:
            text = self.getHTMLdriver(url)
            soup = BeautifulSoup(text, "html.parser")
            a = soup.find_all("a", href=re.compile("path="))
            for aa in a:
                url1 = aa.attrs['href']
                url1 = "http://chromedriver.storage.googleapis.com/index.html" + url1
                dir_name = aa.get_text()
                text1 = self.getHTMLdriver(url1)
                soup1 = BeautifulSoup(text1, "html.parser")
                a1 = soup1.find_all("a")
                for aa1 in a1:
                    url2 = aa1.attrs['href']
                    url2 = "http://chromedriver.storage.googleapis.com" + url2
                    print("url2: " + url2)
                    file_name = aa1.get_text()
                    if file_name.find(".zip", 0, len(file_name)) > 0:
                        self.save(url2, dir_name, file_name)

        except:
            print("except")
user_agent = [
    "Mozilla/5.0 (Macintosh; U; Intel Mac OS X 10_6_8; en-us) AppleWebKit/534.50 (KHTML, like Gecko) Version/5.1 Safari/534.50",
    "Mozilla/5.0 (Windows; U; Windows NT 6.1; en-us) AppleWebKit/534.50 (KHTML, like Gecko) Version/5.1 Safari/534.50",
    "Mozilla/5.0 (Windows NT 10.0; WOW64; rv:38.0) Gecko/20100101 Firefox/38.0",
    "Mozilla/5.0 (Windows NT 10.0; WOW64; Trident/7.0; .NET4.0C; .NET4.0E; .NET CLR 2.0.50727; .NET CLR 3.0.30729; .NET CLR 3.5.30729; InfoPath.3; rv:11.0) like Gecko",
    "Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; Trident/5.0)",
    "Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 6.0; Trident/4.0)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 6.0)",
    "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1)",
    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.6; rv:2.0.1) Gecko/20100101 Firefox/4.0.1",
    "Mozilla/5.0 (Windows NT 6.1; rv:2.0.1) Gecko/20100101 Firefox/4.0.1",
    "Opera/9.80 (Macintosh; Intel Mac OS X 10.6.8; U; en) Presto/2.8.131 Version/11.11",
    "Opera/9.80 (Windows NT 6.1; U; en) Presto/2.8.131 Version/11.11",
    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_7_0) AppleWebKit/535.11 (KHTML, like Gecko) Chrome/17.0.963.56 Safari/535.11",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Maxthon 2.0)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; TencentTraveler 4.0)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; The World)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Trident/4.0; SE 2.X MetaSr 1.0; SE 2.X MetaSr 1.0; .NET CLR 2.0.50727; SE 2.X MetaSr 1.0)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; 360SE)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Avant Browser)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1)",
    "Mozilla/5.0 (iPhone; U; CPU iPhone OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5",
    "Mozilla/5.0 (iPod; U; CPU iPhone OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5",
    "Mozilla/5.0 (iPad; U; CPU OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5",
    "Mozilla/5.0 (Linux; U; Android 2.3.7; en-us; Nexus One Build/FRF91) AppleWebKit/533.1 (KHTML, like Gecko) Version/4.0 Mobile Safari/533.1",
    "MQQBrowser/26 Mozilla/5.0 (Linux; U; Android 2.3.7; zh-cn; MB200 Build/GRJ22; CyanogenMod-7) AppleWebKit/533.1 (KHTML, like Gecko) Version/4.0 Mobile Safari/533.1",
    "Opera/9.80 (Android 2.3.4; Linux; Opera Mobi/build-1107180945; U; en-GB) Presto/2.8.149 Version/11.10",
    "Mozilla/5.0 (Linux; U; Android 3.0; en-us; Xoom Build/HRI39) AppleWebKit/534.13 (KHTML, like Gecko) Version/4.0 Safari/534.13",
    "Mozilla/5.0 (BlackBerry; U; BlackBerry 9800; en) AppleWebKit/534.1+ (KHTML, like Gecko) Version/6.0.0.337 Mobile Safari/534.1+",
    "Mozilla/5.0 (hp-tablet; Linux; hpwOS/3.0.0; U; en-US) AppleWebKit/534.6 (KHTML, like Gecko) wOSBrowser/233.70 Safari/534.6 TouchPad/1.0",
    "Mozilla/5.0 (SymbianOS/9.4; Series60/5.0 NokiaN97-1/20.0.019; Profile/MIDP-2.1 Configuration/CLDC-1.1) AppleWebKit/525 (KHTML, like Gecko) BrowserNG/7.1.18124",
    "Mozilla/5.0 (compatible; MSIE 9.0; Windows Phone OS 7.5; Trident/5.0; IEMobile/9.0; HTC; Titan)",
    "UCWEB7.0.2.37/28/999",
    "NOKIA5700/ UCWEB7.0.2.37/28/999",
    "Openwave/ UCWEB7.0.2.37/28/999",
    "Mozilla/4.0 (compatible; MSIE 6.0; ) Opera/UCWEB7.0.2.37/28/999",
    # iPhone 6：
    "Mozilla/6.0 (iPhone; CPU iPhone OS 8_0 like Mac OS X) AppleWebKit/536.26 (KHTML, like Gecko) Version/8.0 Mobile/10A5376e Safari/8536.25",

]
# hd = {'User-Agent': random.choice(user_agent)}
hd = {'user-agent': 'chorme/10'}

if __name__ == "__main__":
    url = "http://chromedriver.storage.googleapis.com/index.html"
    if not os.path.exists('D:/chromedriver'):  # os模块判断并创建
        os.mkdir('D:/chromedriver/')
    dc = downloadchromedriver(hd)
    dc.download(url)
